from rest_framework.views import APIView
from rest_framework_jwt.settings import api_settings

from .utils import OAuthQQ
from rest_framework import status
from rest_framework.generics import GenericAPIView #有两个模型类
from rest_framework.response import Response
from .exceptions import QQAPIException
from .models import OAuthQQUser


#(2.1)扫码成功后的请求响应 get
#(2.2)第一次登陆本站时候进行的表单提交  post
class QQAuthUserView(GenericAPIView):
    # 获取扫码成功后的code
    # 根据code获取access_token
    # 根据access_token获取openid(用户信息)
    # 查询本openid是否绑定过本网站
    # 如果绑定过则生成JWT token并返回
    # 如果没绑定过,创建用户进行绑定
    def get(self,request):
        code = request.query_params.get('code')
        print('(2.1)扫码成功后qq服务器赋予的code：',code)
        if not code:
            return Response({'message':'缺少code'},status=status.HTTP_400_BAD_REQUEST)

        oauth = OAuthQQ()
        try:
            access_token = oauth.get_access_token(code)
            openid = oauth.get_openid(access_token)
        except QQAPIException:
            return Response({'message': 'QQ服务异常'}, status=status.HTTP_503_SERVICE_UNAVAILABLE)
        print('(2.2)使用code向qq服务器请求到的access_token：', access_token)
        print('(2.3)使用access_token向qq服务器请求到的openid：', openid)

        try:
            oauth_user = OAuthQQUser.objects.get(openid=openid)
        except OAuthQQUser.DoesNotExist:
            # (2.11)抛出异常说明该openid没有绑定到本站,生成一个状态保持的安全的token给前端
            # 不能直接将这个Openid返回给前端，调用工具里的加密签名的方法，返回一个安全的openid
            # 这个安全的Openid就是access_token
            access_token_openid = OAuthQQ.generate_save_user_token(openid)
            return Response({'access_token': access_token_openid})
        else:
            # 如果openid已绑定美多商城用户，直接生成JWT token，并返回
            user = oauth_user.user

            # 生成JWT——token
            jwt_payload_handler = api_settings.JWT_PAYLOAD_HANDLER
            jwt_encode_handler = api_settings.JWT_ENCODE_HANDLER

            payload = jwt_payload_handler(user)
            token = jwt_encode_handler(payload)

            return Response({
                'token': token,
                'user_id': user.id,
                'username': user.username
            })

    # (2.12)如果openid没有绑定美多商城用户，跳转到绑定页面，进行一次表单提交
    #
    def post(self, request):
        """绑定用户到openid"""
        # 获取序列化器:并向序列化器内部传入POST请求发送过来的请求体
        serializer = self.get_serializer(data=request.data)

        # 开启校验
        serializer.is_valid(raise_exception=True)

        # 保存校验后的数据:序列化器中的create方法返回校验后的user
        user = serializer.save()

        # 再次生成JWT token
        jwt_payload_handler = api_settings.JWT_PAYLOAD_HANDLER
        jwt_encode_handler = api_settings.JWT_ENCODE_HANDLER

        payload = jwt_payload_handler(user)
        token = jwt_encode_handler(payload)

        return Response({
            'token': token,
            'user_id': user.id,
            'username': user.username
        })


# (1)点击QQ登录标签发起的请求
# 点击后url指向此视图函数，视图函数获取next，调用工具中的合成url方法，将qq登陆的url返回给浏览器
class QQAuthURLView(APIView):
    """提供QQ登录界面的URL"""

    def get(self, request):

        # 获取next
        next = request.query_params.get('next')
        print('(1.1)点击qq登陆按钮后的next:',next)

        # 创建oauth工具对象
        oauth = OAuthQQ(state=next)

        # 调用获取QQ登录界面的URL的工具方法
        login_url =  oauth.get_qq_login_url()
        print('(1.2)服务器返回合成的QQ登录界面的URL：',login_url)

        # 返回QQ登录界面的URL给浏览器
        return Response({'login_url':login_url})
